package com.study.dataStructure.sparseArray;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;

/**
 * @PACKAGE_NAME: com.study.dataStructure.sparsearr
 * @CLASS_NAME: SparseArray
 * @USER: kagura
 * @Description TODO 稀疏数组的使用
 * @Date 2021/8/21 6:53 下午
 * @MONTH_NAME_FULL: 八月
 * @DAY: 21
 * @DAY_NAME_FULL: 星期六
 * @PROJECT_NAME: Study
 **/
public class SparseArray {
    public static void main(String[] args) throws IOException {
        // 创建一个原始的二维数组11*11
        // 0:没有棋子，1黑子，2蓝字
        int chessArray1[][] = new int[11][11];
        chessArray1[1][2] = 1;
        chessArray1[2][3] = 2;
        chessArray1[3][4] = 1;
        chessArray1[4][5] = 2;
        // 输出原来的二维数组
        System.out.println("原始的二维数组: ");
        for (int[] row : chessArray1) {
            for(int data : row) {
                System.out.printf("%d\t", data);
            }
            System.out.println();
        }
        // 将二维数组转为稀疏数组
        // 1.先遍历二维数组得到非0数据的个数
        int sum = 0;
        for (int i = 0; i < 11; i++) {
            for (int j = 0; j < 11; j++) {
                if (chessArray1[i][j] != 0) {
                    sum ++;
                }
            }
        }
        // System.out.println(sum);
        // 创建对应的稀疏数组
        int sparseArr[][] = new int[sum+1][3];
        // 给稀疏数组赋值
        sparseArr[0][0] = 11;
        sparseArr[0][1] = 11;
        sparseArr[0][2] = sum;

        // 遍历二维数组，将非0的值存放到稀疏数组中
        int count = 0; // 用于记录是第几个非0数据
        for (int i = 0; i < 11; i++) {
            for (int j = 0; j < 11; j++) {
                if (chessArray1[i][j] != 0) {
                    count++;
                    sparseArr[count][0] = i;
                    sparseArr[count][1] = j;
                    sparseArr[count][2] = chessArray1[i][j];
                }
            }
        }
        // 输出稀疏数组的形式
        System.out.println();
        System.out.println("得到的稀疏数组为: ");
        for (int i = 0; i < sparseArr.length; i++) {
            System.out.printf("%d\t%d\t%d\t\n", sparseArr[i][0], sparseArr[i][1], sparseArr[i][2]);
        }
        System.out.println();

        // 存储到本地
        File file = new File("resources/1.data");
        FileWriter out = new FileWriter(file);
        for (int i = 0; i < 5; i++) {
            for (int j = 0; j < 3; j++) {
                out.write(sparseArr[i][j] + "\t");
            }
            out.write("\r\n");
        }
        out.close();

        // 将稀疏数组恢复成原始的二维数组
        // 1. 先读取稀疏数组第一行，得到行列数
        int chessArray2[][] = new int[sparseArr[0][0]][sparseArr[0][1]];

        // 2. 把读取稀疏数组后几行的数据(从第二行开始)，并赋值给原始的二维数组
        for (int i = 1; i < sparseArr.length; i++) {
            chessArray2[sparseArr[i][0]][sparseArr[i][1]] = sparseArr[i][2];
        }

        // 恢复后的二维数组
        System.out.println("恢复的二维数组: ");
        for (int[] row : chessArray2) {
            for(int data : row) {
                System.out.printf("%d\t", data);
            }
            System.out.println();
        }
    }
}
